package com.atguigu.rabbitmq.four;

import com.atguigu.rabbitmq.utils.RabbitmqUtils;
import com.rabbitmq.client.Channel;
import com.rabbitmq.client.ConfirmCallback;

import java.util.UUID;
import java.util.concurrent.ConcurrentNavigableMap;
import java.util.concurrent.ConcurrentSkipListMap;

/**
 * 发布确认模式
 * 单个确认发布
 * 批量确认发布
 * 异步确认发布
 */
public class ConfirmMessage {

    // 批量发消息的个数
    public static final int MESSAGE_COUNT = 1000;

    public static void main(String[] args) throws Exception {
        // 1. 单个确认
        //publishMessageSimple(); // 590毫秒
        // 2. 批量确认
        //publishMessageBatch();  // 126毫秒
        // 3. 异步确认
        publishMessageAsync();  // 70毫秒
    }

    public static void publishMessageAsync() throws Exception{
        Channel channel = RabbitmqUtils.getChannel();
        // 队列声明
        String queueName = UUID.randomUUID().toString();
        // 创建队列
        channel.queueDeclare(queueName,false,false,false,null);
        //开启发布确认
        channel.confirmSelect();
        /**
         * 线程安全有序的一个哈希表 适用于高并发的情况下
         * 1. 能轻松的将序号与消息进行关联
         * 2. 轻松的批量删除条目 只要给到序号
         * 3. 支持高并发（多线程）
         */
        ConcurrentSkipListMap<Long,String> outstandingConfirms = new ConcurrentSkipListMap<>();

        // 消息确认成功回调的函数
        ConfirmCallback ackCallback = (deliveryTag,multiple)->{
            if(multiple){ //一般没有批量确认
                // 删除掉已经确认的消息
                ConcurrentNavigableMap<Long,String> confirmed = outstandingConfirms.headMap(deliveryTag);
                confirmed.clear();
            }else{
                outstandingConfirms.remove(deliveryTag);
            }

            System.out.println("确认消息：" + deliveryTag);
        };
        // 消息确认失败回调的函数
        /**
         * 1. 消息的标记
         * 2. 是否为批量确认
         */
        ConfirmCallback nackCallback= (deliveryTag,multiple)->{
            // 打印下未确认的消息有哪些
            String message = outstandingConfirms.get(deliveryTag);
            System.out.println("未确认消息是：" + message+"；未确认消息的标记是：" + deliveryTag);
        };
        // 消息的监听器，监听哪些消息成功了，哪些消息失败了
        /**
         * 1. 监听哪些消息成功了
         * 2. 监听哪些消息失败了
         */
        channel.addConfirmListener(ackCallback,nackCallback); //异步通知
        // 开始的时间 毫秒
        Long startTime = System.currentTimeMillis();
        //批量的发消息
        for (int i = 0; i < MESSAGE_COUNT; i++) {
            String message = i + "";
            channel.basicPublish("",queueName,null,message.getBytes());
            // 记录下所有要发送的消息
            outstandingConfirms.put(channel.getNextPublishSeqNo(),message);
        }
        // 结束的时间 毫秒
        Long endTime = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_COUNT+"个异步确认发布消息，耗时" + ((endTime - startTime)) + "毫秒");
    }

    public static void publishMessageBatch() throws Exception{
        Channel channel = RabbitmqUtils.getChannel();
        // 队列声明
        String queueName = UUID.randomUUID().toString();
        // 持久化队列
        channel.queueDeclare(queueName,true,false,false,null);
        //开启发布确认
        channel.confirmSelect();
        // 开始的时间 毫秒
        Long startTime = System.currentTimeMillis();
        // 批量确认消息的大小
        int batchSize = 100;
        // 消息发送次数记录
        int messageStep = 0;
        //批量的发消息
        for (int i = 0; i < MESSAGE_COUNT; i++) {
            String message = i + "";
            channel.basicPublish("",queueName,null,message.getBytes());
            // 判断达到一百条，批量确认
            if(messageStep == batchSize){
                messageStep = 0;
                channel.waitForConfirms();
            }else{
                messageStep++;
            }
        }
        // 结束的时间 毫秒
        Long endTime = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_COUNT+"个批量确认发布消息，耗时" + ((endTime - startTime)) + "毫秒");
    }

    /**
     * 单个确认发布
     */
    public static void publishMessageSimple() throws Exception{
        Channel channel = RabbitmqUtils.getChannel();
        // 队列声明
        String queueName = UUID.randomUUID().toString();
        // 持久化队列
        channel.queueDeclare(queueName,true,false,false,null);
        //开启发布确认
        channel.confirmSelect();
        // 开始的时间 毫秒
        Long startTime = System.currentTimeMillis();
        //批量的发消息
        for (int i = 0; i < MESSAGE_COUNT; i++) {
            String message = i + "";
            channel.basicPublish("",queueName,null,message.getBytes());
            // 单个消息就进行发布确认
            boolean flag = channel.waitForConfirms();
            if(flag){
                System.out.println("第"+(i+1)+"次消息发送成功");
            }
        }
        // 结束的时间 毫秒
        Long endTime = System.currentTimeMillis();
        System.out.println("发布"+MESSAGE_COUNT+"个单独确认发布消息，耗时" + ((endTime - startTime)) + "毫秒");

    }
}
